import sys
import time
import pygame
from pygame.locals import *
from airplane_sprites import *


class AirplaneGame:
    """飞机大战主程序"""

    def __init__(self):
        # 游戏初始化 TODO
        print('游戏初始化')
        # 1、创建游戏窗口
        self.screen = pygame.display.set_mode(SCREEN_RECT.size)
        pygame.display.set_caption('V1.0.0 飞机大战游戏')
        # 2、创建时钟对象
        self.clock = pygame.time.Clock()
        # 3、创建精灵、精灵组对象
        self.__create_sprite()
        # 4、设置定时器时间-创建敌机-时间间隔1s
        pygame.time.set_timer(CREATE_ENEMY_EVENT, 1000)
        # 4、设置定时器时间-英雄发射子弹-时间间隔0.5s
        pygame.time.set_timer(HERO_FIRE_EVENT, 500)
        print('游戏初始化完成')

    def __create_sprite(self):
        # 创建精灵或者精灵组 TODO
        # 1、创建背景精灵和精灵组
        bg1 = Background(True)
        bg2 = Background()
        self.back_group = pygame.sprite.Group(bg1, bg2)
        # 2、创建敌机精灵组
        self.enemy_group = pygame.sprite.Group()
        # 3、创建英雄精灵和精灵组
        self.hero = Hero()
        self.hero_group = pygame.sprite.Group(self.hero)
        pass

    def start_game(self):
        # 开始游戏 TODO
        print('游戏开始……')
        while True:
            # 1、设置刷新帧率60/s
            self.clock.tick(FRAME_RATE)
            # 2、设置事件监听
            self.__event_handler()
            # 3、碰撞检测
            self.__check_collide()
            # 4、精灵组更新/绘制
            self.__update_sprites()
            # 5、更新显示
            pygame.display.update()

    def __event_handler(self):
        # 事件监听 TODO
        for event in pygame.event.get():
            # 判断是否退出游戏
            if event.type == pygame.QUIT:
                AirplaneGame.__game_over()
            elif event.type == CREATE_ENEMY_EVENT:
                self.enemy_group.add(Enemy())
            elif event.type == HERO_FIRE_EVENT:
                self.hero.fire()
        # 使用键盘模块提供的方法获取键盘按键事件
        keys_pressed = pygame.key.get_pressed()
        # 如果键盘按压事件元组对应按键则值为1
        if keys_pressed[pygame.K_LEFT]:
            # print("向左移动……")
            self.hero.adjust_speed(-5)
        elif keys_pressed[pygame.K_RIGHT]:
            # print("向右移动……")
            self.hero.adjust_speed(5)
        else:
            self.hero.adjust_speed()
        pass

    def __check_collide(self):
        # 碰撞检测 TODO
        # 1、子弹摧毁敌机：子弹精灵组与敌机精灵组碰撞检测
        pygame.sprite.groupcollide(self.enemy_group, self.hero.get_bullet_group(), True, True)
        # 2、敌机摧毁英雄：英雄与敌机精灵组碰撞检测
        enemies = pygame.sprite.spritecollide(self.hero, self.enemy_group, True)
        # 检测敌机列表
        if len(enemies) > 0:
            # 销毁英雄
            self.hero.kill()
            # 结束游戏
            time.sleep(1)
            AirplaneGame.__game_over()
        pass

    def __update_sprites(self):
        # 精灵组更新/绘制 TODO
        # 1、背景精灵组更新/绘制
        self.back_group.update()
        self.back_group.draw(self.screen)
        # 2、敌机精灵组更新/绘制
        self.enemy_group.update()
        self.enemy_group.draw(self.screen)
        # 3、英雄精灵组绘制/更新
        self.hero_group.update()
        self.hero_group.draw(self.screen)
        # 4、调用英雄绘制/更新子弹精灵组方法
        self.hero.update_bullet_group(self.screen)
        pass

    @staticmethod
    def __game_over():
        # 游戏结束
        print('Game Over!')
        pygame.quit()  # pygame清理
        sys.exit()  # 程序结束


if __name__ == '__main__':
    # 创建游戏对象
    game = AirplaneGame()
    # 开始游戏
    game.start_game()
